Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add gc and fix typo #202

Closed
wants to merge 1 commit into from
Closed

Add gc and fix typo #202

wants to merge 1 commit into from

Conversation

BeingGod
Copy link

Pull Request Template

Issue Reference

N/A

Description

Adding gc collect and torch.cuda.empty_cache() after del to make sure device memory is freed immediately.

Type of Change

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • This change requires a documentation update

How Has This Been Tested?

N/A

Checklist

  • I have read the CONTRIBUTING document.
  • My code follows the code style of this project.
  • My change requires a change to the documentation.
  • I have updated the documentation accordingly.
  • I have added tests to cover my changes.
  • All new and existing tests passed.

Additional Information

N/A

@BeingGod
add gc and fix typo
a956a8d 
REASON: Adding gc collect and torch.cuda.empty_cache() after del to make sure device memory is freed immediately
@BeingGod
Copy link
Author

PTAL @MayDomine

@BeingGod BeingGod closed this by deleting the head repository Oct 14, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@BeingGod